from predict import HandWrittenDigitRecognizer

class Model(object):
    def __init__(self) -> None:
        self.presenter = None

        modelPath = "./model/model.h5"
        self.predicter = HandWrittenDigitRecognizer(modelPath)

    def setPresenter(self, presenter):
        self.presenter = presenter

    def getLabelList(self, tragetImageList):
        return self.predicter.getLabelList(tragetImageList)